import qrcode
import os


__all__ = [
    "generate_qrcode",
    "delete_qrcode",
]


def generate_qrcode(code, img_path):
    # 创建一个二维码对象
    qr = qrcode.QRCode(
        version=1,  # 控制二维码的大小，1 是最小的
        error_correction=qrcode.constants.ERROR_CORRECT_L,  # 错误校正级别
        box_size=10,  # 每个方块的像素大小
        border=1,  # 边框的宽度
    )
    # 添加数据到二维码
    qr.add_data(code)
    qr.make(fit=True)

    # 创建一个图片对象
    img = qr.make_image(fill_color="black", back_color="transparent")
    # 保存图片
    img.save(img_path)

def delete_qrcode(img_path):
    if os.path.exists(img_path):
        os.remove(img_path)